from pydantic import BaseModel
from typing import Literal


class MCPCreateSchema(BaseModel):
    name: str
    transport: Literal['streamable_http', 'sse', 'stdio']
    url: str|None = None
    headers: str|None = None
    command: str|None = None
    args: str|None = None
    env: str|None = None
